package com.jdjc.subject.infra.basic.utils;

import com.alibaba.druid.filter.config.ConfigTools;

import java.security.NoSuchAlgorithmException;
import java.security.NoSuchProviderException;

/**
 * 数据库加密
 * 使用 ConfigTools 类生成一对公钥和私钥。
 * <p>
 * 使用私钥加密明文（例如，数据库密码）。
 * <p>
 * 在配置文件中存储加密后的文本（密文）。
 * <p>
 * 在需要时使用公钥将密文解密回明文。
 */
public class DruidEncrypUtil {
    private static String publicKey;
    private static String privateKey;

    static {
        try {
            //使用 ConfigTools 类生成一对公钥和私钥。
            String[] keyPair = ConfigTools.genKeyPair(512);
            privateKey = keyPair[0];
            System.out.println("privateKey:" + privateKey);

            publicKey = keyPair[1];
            System.out.println("publicKey:" + publicKey);
        } catch (NoSuchAlgorithmException e) {
            throw new RuntimeException(e);
        } catch (NoSuchProviderException e) {
            throw new RuntimeException(e);
        }
    }

    /**
     * 加密
     *
     * @param plainText
     * @return
     */
    public static String encrypt(String plainText) throws Exception {

        String encrypt = ConfigTools.encrypt(privateKey, plainText);
        System.out.println("encrypt:" + encrypt);

        return encrypt;
    }

    /**
     * 解密
     *
     * @param plainText
     * @return
     */
    public static String decrypt(String plainText) throws Exception {

        String decrypt = ConfigTools.encrypt(publicKey, plainText);
        System.out.println("decrypt:" + decrypt);

        return decrypt;
    }

    public static void main(String[] args) throws Exception {
        String encrypt= encrypt("123");
        System.out.println("encrypt:" + encrypt);
    }
}
